/*M///////////////////////////////////////////////////////////////////////////////////////
//
//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
//
//  By downloading, copying, installing or using the software you agree to this license.
//  If you do not agree to this license, do not download, install,
//  copy or use the software.
//
//
//                        Intel License Agreement
//                For Open Source Computer Vision Library
//
// Copyright (C) 2000, Intel Corporation, all rights reserved.
// Third party copyrights are property of their respective owners.
//
// Redistribution and use in source and binary forms, with or without modification,
// are permitted provided that the following conditions are met:
//
//   * Redistribution's of source code must retain the above copyright notice,
//     this list of conditions and the following disclaimer.
//
//   * Redistribution's in binary form must reproduce the above copyright notice,
//     this list of conditions and the following disclaimer in the documentation
//     and/or other materials provided with the distribution.
//
//   * The name of Intel Corporation may not be used to endorse or promote products
//     derived from this software without specific prior written permission.
//
// This software is provided by the copyright holders and contributors "as is" and
// any express or implied warranties, including, but not limited to, the implied
// warranties of merchantability and fitness for a particular purpose are disclaimed.
// In no event shall the Intel Corporation or contributors be liable for any direct,
// indirect, incidental, special, exemplary, or consequential damages
// (including, but not limited to, procurement of substitute goods or services;
// loss of use, data, or profits; or business interruption) however caused
// and on any theory of liability, whether in contract, strict liability,
// or tort (including negligence or otherwise) arising in any way out of
// the use of this software, even if advised of the possibility of such damage.
//
//M*/

#include "precomp.hpp"
#include "grfmt_jpeg.hpp"

#ifdef HAVE_JPEG

#include <stdio.h>
#include <setjmp.h>

#ifdef WIN32

#define XMD_H // prevent redefinition of INT32
#undef FAR  // prevent FAR redefinition

#endif

#if defined WIN32 && defined __GNUC__
typedef unsigned char boolean;
#endif

extern "C" {
#include "jpeglib.h"
}

namespace cv {

struct JpegErrorMgr {
	struct jpeg_error_mgr pub;
	jmp_buf setjmp_buffer;
};

struct JpegSource {
	struct jpeg_source_mgr pub;
	int skip;
};

struct JpegState {
	jpeg_decompress_struct cinfo; // IJG JPEG codec structure
	JpegErrorMgr jerr; // error processing manager state
	JpegSource source; // memory buffer source
};

/////////////////////// Error processing /////////////////////

METHODDEF(void)
stub(j_decompress_ptr) {
}

METHODDEF(boolean)
fill_input_buffer(j_decompress_ptr) {
	return FALSE;
}

// emulating memory input stream

METHODDEF(void)
skip_input_data(j_decompress_ptr cinfo, long num_bytes) {
	JpegSource* source = (JpegSource*) cinfo->src;

	if ( num_bytes > (long)source->pub.bytes_in_buffer ) {
		// We need to skip more data than we have in the buffer.
		// This will force the JPEG library to suspend decoding.
		source->skip = num_bytes - source->pub.bytes_in_buffer;
		source->pub.next_input_byte += source->pub.bytes_in_buffer;
		source->pub.bytes_in_buffer = 0;
	} else {
		// Skip portion of the buffer
		source->pub.bytes_in_buffer -= num_bytes;
		source->pub.next_input_byte += num_bytes;
		source->skip = 0;
	}
}


GLOBAL(void)
jpeg_buffer_src(j_decompress_ptr cinfo, JpegSource* source) {
	cinfo->src = &source->pub;

	// Prepare for suspending reader
	source->pub.init_source = stub;
	source->pub.fill_input_buffer = fill_input_buffer;
	source->pub.skip_input_data = skip_input_data;
	source->pub.resync_to_restart = jpeg_resync_to_restart;
	source->pub.term_source = stub;
	source->pub.bytes_in_buffer = 0; // forces fill_input_buffer on first read

	source->skip = 0;
}


METHODDEF(void)
error_exit( j_common_ptr cinfo ) {
	JpegErrorMgr* err_mgr = (JpegErrorMgr*)(cinfo->err);

	/* Return control to the setjmp point */
	longjmp( err_mgr->setjmp_buffer, 1 );
}


/////////////////////// JpegDecoder ///////////////////


JpegDecoder::JpegDecoder() {
	m_signature = "\xFF\xD8\xFF";
	m_state = 0;
	m_f = 0;
	m_buf_supported = true;
}


JpegDecoder::~JpegDecoder() {
	close();
}


void  JpegDecoder::close() {
	if ( m_state ) {
		JpegState* state = (JpegState*)m_state;
		jpeg_destroy_decompress( &state->cinfo );
		delete state;
		m_state = 0;
	}

	if ( m_f ) {
		fclose( m_f );
		m_f = 0;
	}

	m_width = m_height = 0;
	m_type = -1;
}

ImageDecoder JpegDecoder::newDecoder() const {
	return new JpegDecoder;
}

bool  JpegDecoder::readHeader() {
	bool result = false;
	close();

	JpegState* state = new JpegState;
	m_state = state;
	state->cinfo.err = jpeg_std_error(&state->jerr.pub);
	state->jerr.pub.error_exit = error_exit;

	if ( setjmp( state->jerr.setjmp_buffer ) == 0 ) {
		jpeg_create_decompress( &state->cinfo );

		if ( !m_buf.empty() ) {
			jpeg_buffer_src(&state->cinfo, &state->source);
			state->source.pub.next_input_byte = m_buf.data;
			state->source.pub.bytes_in_buffer = m_buf.cols * m_buf.rows * m_buf.elemSize();
		} else {
			m_f = fopen( m_filename.c_str(), "rb" );
			if ( m_f ) {
				jpeg_stdio_src( &state->cinfo, m_f );
			}
		}
		jpeg_read_header( &state->cinfo, TRUE );

		m_width = state->cinfo.image_width;
		m_height = state->cinfo.image_height;
		m_type = state->cinfo.num_components > 1 ? CV_8UC3 : CV_8UC1;
		result = true;
	}

	if ( !result ) {
		close();
	}

	return result;
}

/***************************************************************************
 * following code is for supporting MJPEG image files
 * based on a message of Laurent Pinchart on the video4linux mailing list
 ***************************************************************************/

/* JPEG DHT Segment for YCrCb omitted from MJPEG data */
static
unsigned char my_jpeg_odml_dht[0x1a4] = {
	0xff, 0xc4, 0x01, 0xa2,

	0x00, 0x00, 0x01, 0x05, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x00, 0x00,
	0x00, 0x00, 0x00, 0x00, 0x00,
	0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b,

	0x01, 0x00, 0x03, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01,
	0x00, 0x00, 0x00, 0x00, 0x00,
	0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b,

	0x10, 0x00, 0x02, 0x01, 0x03, 0x03, 0x02, 0x04, 0x03, 0x05, 0x05, 0x04,
	0x04, 0x00, 0x00, 0x01, 0x7d,
	0x01, 0x02, 0x03, 0x00, 0x04, 0x11, 0x05, 0x12, 0x21, 0x31, 0x41, 0x06,
	0x13, 0x51, 0x61, 0x07,
	0x22, 0x71, 0x14, 0x32, 0x81, 0x91, 0xa1, 0x08, 0x23, 0x42, 0xb1, 0xc1,
	0x15, 0x52, 0xd1, 0xf0,
	0x24, 0x33, 0x62, 0x72, 0x82, 0x09, 0x0a, 0x16, 0x17, 0x18, 0x19, 0x1a,
	0x25, 0x26, 0x27, 0x28,
	0x29, 0x2a, 0x34, 0x35, 0x36, 0x37, 0x38, 0x39, 0x3a, 0x43, 0x44, 0x45,
	0x46, 0x47, 0x48, 0x49,
	0x4a, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, 0x59, 0x5a, 0x63, 0x64, 0x65,
	0x66, 0x67, 0x68, 0x69,
	0x6a, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78, 0x79, 0x7a, 0x83, 0x84, 0x85,
	0x86, 0x87, 0x88, 0x89,
	0x8a, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98, 0x99, 0x9a, 0xa2, 0xa3,
	0xa4, 0xa5, 0xa6, 0xa7,
	0xa8, 0xa9, 0xaa, 0xb2, 0xb3, 0xb4, 0xb5, 0xb6, 0xb7, 0xb8, 0xb9, 0xba,
	0xc2, 0xc3, 0xc4, 0xc5,
	0xc6, 0xc7, 0xc8, 0xc9, 0xca, 0xd2, 0xd3, 0xd4, 0xd5, 0xd6, 0xd7, 0xd8,
	0xd9, 0xda, 0xe1, 0xe2,
	0xe3, 0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9, 0xea, 0xf1, 0xf2, 0xf3, 0xf4,
	0xf5, 0xf6, 0xf7, 0xf8,
	0xf9, 0xfa,

	0x11, 0x00, 0x02, 0x01, 0x02, 0x04, 0x04, 0x03, 0x04, 0x07, 0x05, 0x04,
	0x04, 0x00, 0x01, 0x02, 0x77,
	0x00, 0x01, 0x02, 0x03, 0x11, 0x04, 0x05, 0x21, 0x31, 0x06, 0x12, 0x41,
	0x51, 0x07, 0x61, 0x71,
	0x13, 0x22, 0x32, 0x81, 0x08, 0x14, 0x42, 0x91, 0xa1, 0xb1, 0xc1, 0x09,
	0x23, 0x33, 0x52, 0xf0,
	0x15, 0x62, 0x72, 0xd1, 0x0a, 0x16, 0x24, 0x34, 0xe1, 0x25, 0xf1, 0x17,
	0x18, 0x19, 0x1a, 0x26,
	0x27, 0x28, 0x29, 0x2a, 0x35, 0x36, 0x37, 0x38, 0x39, 0x3a, 0x43, 0x44,
	0x45, 0x46, 0x47, 0x48,
	0x49, 0x4a, 0x53, 0x54, 0x55, 0x56, 0x57, 0x58, 0x59, 0x5a, 0x63, 0x64,
	0x65, 0x66, 0x67, 0x68,
	0x69, 0x6a, 0x73, 0x74, 0x75, 0x76, 0x77, 0x78, 0x79, 0x7a, 0x82, 0x83,
	0x84, 0x85, 0x86, 0x87,
	0x88, 0x89, 0x8a, 0x92, 0x93, 0x94, 0x95, 0x96, 0x97, 0x98, 0x99, 0x9a,
	0xa2, 0xa3, 0xa4, 0xa5,
	0xa6, 0xa7, 0xa8, 0xa9, 0xaa, 0xb2, 0xb3, 0xb4, 0xb5, 0xb6, 0xb7, 0xb8,
	0xb9, 0xba, 0xc2, 0xc3,
	0xc4, 0xc5, 0xc6, 0xc7, 0xc8, 0xc9, 0xca, 0xd2, 0xd3, 0xd4, 0xd5, 0xd6,
	0xd7, 0xd8, 0xd9, 0xda,
	0xe2, 0xe3, 0xe4, 0xe5, 0xe6, 0xe7, 0xe8, 0xe9, 0xea, 0xf2, 0xf3, 0xf4,
	0xf5, 0xf6, 0xf7, 0xf8,
	0xf9, 0xfa
};

/*
 * Parse the DHT table.
 * This code comes from jpeg6b (jdmarker.c).
 */
static
int my_jpeg_load_dht (struct jpeg_decompress_struct* info, unsigned char* dht,
					  JHUFF_TBL* ac_tables[], JHUFF_TBL* dc_tables[]) {
	unsigned int length = (dht[2] << 8) + dht[3] - 2;
	unsigned int pos = 4;
	unsigned int count, i;
	int index;

	JHUFF_TBL** hufftbl;
	unsigned char bits[17];
	unsigned char huffval[256];

	while (length > 16) {
		bits[0] = 0;
		index = dht[pos++];
		count = 0;
		for (i = 1; i <= 16; ++i) {
			bits[i] = dht[pos++];
			count += bits[i];
		}
		length -= 17;

		if (count > 256 || count > length) {
			return -1;
		}

		for (i = 0; i < count; ++i) {
			huffval[i] = dht[pos++];
		}
		length -= count;

		if (index & 0x10) {
			index -= 0x10;
			hufftbl = &ac_tables[index];
		} else {
			hufftbl = &dc_tables[index];
		}

		if (index < 0 || index >= NUM_HUFF_TBLS) {
			return -1;
		}

		if (*hufftbl == NULL) {
			*hufftbl = jpeg_alloc_huff_table ((j_common_ptr)info);
		}
		if (*hufftbl == NULL) {
			return -1;
		}

		memcpy ((*hufftbl)->bits, bits, sizeof (*hufftbl)->bits);
		memcpy ((*hufftbl)->huffval, huffval, sizeof (*hufftbl)->huffval);
	}

	if (length != 0) {
		return -1;
	}

	return 0;
}

/***************************************************************************
 * end of code for supportting MJPEG image files
 * based on a message of Laurent Pinchart on the video4linux mailing list
 ***************************************************************************/

bool  JpegDecoder::readData( Mat& img ) {
	bool result = false;
	uchar* data = img.data;
	int step = img.step;
	bool color = img.channels() > 1;
	JpegState* state = (JpegState*)m_state;

	if ( state && m_width && m_height ) {
		jpeg_decompress_struct* cinfo = &state->cinfo;
		JpegErrorMgr* jerr = &state->jerr;
		JSAMPARRAY buffer = 0;

		if ( setjmp( jerr->setjmp_buffer ) == 0 ) {
			/* check if this is a mjpeg image format */
			if ( cinfo->ac_huff_tbl_ptrs[0] == NULL &&
					cinfo->ac_huff_tbl_ptrs[1] == NULL &&
					cinfo->dc_huff_tbl_ptrs[0] == NULL &&
					cinfo->dc_huff_tbl_ptrs[1] == NULL ) {
				/* yes, this is a mjpeg image format, so load the correct
				huffman table */
				my_jpeg_load_dht( cinfo,
								  my_jpeg_odml_dht,
								  cinfo->ac_huff_tbl_ptrs,
								  cinfo->dc_huff_tbl_ptrs );
			}

			if ( color ) {
				if ( cinfo->num_components != 4 ) {
					cinfo->out_color_space = JCS_RGB;
					cinfo->out_color_components = 3;
				} else {
					cinfo->out_color_space = JCS_CMYK;
					cinfo->out_color_components = 4;
				}
			} else {
				if ( cinfo->num_components != 4 ) {
					cinfo->out_color_space = JCS_GRAYSCALE;
					cinfo->out_color_components = 1;
				} else {
					cinfo->out_color_space = JCS_CMYK;
					cinfo->out_color_components = 4;
				}
			}

			jpeg_start_decompress( cinfo );

			buffer = (*cinfo->mem->alloc_sarray)((j_common_ptr)cinfo,
												 JPOOL_IMAGE, m_width * 4, 1 );

			for ( ; m_height--; data += step ) {
				jpeg_read_scanlines( cinfo, buffer, 1 );
				if ( color ) {
					if ( cinfo->out_color_components == 3 ) {
						icvCvt_RGB2BGR_8u_C3R( buffer[0], 0, data, 0, cvSize(m_width, 1) );
					} else {
						icvCvt_CMYK2BGR_8u_C4C3R( buffer[0], 0, data, 0, cvSize(m_width, 1) );
					}
				} else {
					if ( cinfo->out_color_components == 1 ) {
						memcpy( data, buffer[0], m_width );
					} else {
						icvCvt_CMYK2Gray_8u_C4C1R( buffer[0], 0, data, 0, cvSize(m_width, 1) );
					}
				}
			}
			result = true;
			jpeg_finish_decompress( cinfo );
		}
	}

	close();
	return result;
}


/////////////////////// JpegEncoder ///////////////////

struct JpegDestination {
	struct jpeg_destination_mgr pub;
	vector<uchar> *buf, *dst;
};

METHODDEF(void)
stub(j_compress_ptr) {
}

METHODDEF(void)
term_destination (j_compress_ptr cinfo) {
	JpegDestination* dest = (JpegDestination*)cinfo->dest;
	size_t sz = dest->dst->size(), bufsz = dest->buf->size() - dest->pub.free_in_buffer;
	if ( bufsz > 0 ) {
		dest->dst->resize(sz + bufsz);
		memcpy( &(*dest->dst)[0] + sz, &(*dest->buf)[0], bufsz);
	}
}

METHODDEF(boolean)
empty_output_buffer (j_compress_ptr cinfo) {
	JpegDestination* dest = (JpegDestination*)cinfo->dest;
	size_t sz = dest->dst->size(), bufsz = dest->buf->size();
	dest->dst->resize(sz + bufsz);
	memcpy( &(*dest->dst)[0] + sz, &(*dest->buf)[0], bufsz);

	dest->pub.next_output_byte = &(*dest->buf)[0];
	dest->pub.free_in_buffer = bufsz;
	return TRUE;
}

GLOBAL(void)
jpeg_buffer_dest(j_compress_ptr cinfo, JpegDestination* destination) {
	cinfo->dest = &destination->pub;

	destination->pub.init_destination = stub;
	destination->pub.empty_output_buffer = empty_output_buffer;
	destination->pub.term_destination = term_destination;
}


JpegEncoder::JpegEncoder() {
	m_description = "JPEG files (*.jpeg;*.jpg;*.jpe)";
	m_buf_supported = true;
}


JpegEncoder::~JpegEncoder() {
}

ImageEncoder JpegEncoder::newEncoder() const {
	return new JpegEncoder;
}

bool  JpegEncoder::write( const Mat& img, const vector<int>& params ) {
	int quality = 95;

	for ( size_t i = 0; i < params.size(); i += 2 ) {
		if ( params[i] == CV_IMWRITE_JPEG_QUALITY ) {
			quality = params[i+1];
			quality = MIN(MAX(quality, 0), 100);
		}
	}

	bool result = false;
	FILE* f = 0;
	int _channels = img.channels();
	int channels = _channels > 1 ? 3 : 1;
	int width = img.cols, height = img.rows;

	vector<uchar> out_buf(1 << 12);
	AutoBuffer<uchar> _buffer;
	uchar* buffer;

	struct jpeg_compress_struct cinfo;
	JpegErrorMgr jerr;
	JpegDestination dest;

	jpeg_create_compress(&cinfo);
	cinfo.err = jpeg_std_error(&jerr.pub);
	jerr.pub.error_exit = error_exit;

	if ( !m_buf ) {
		f = fopen( m_filename.c_str(), "wb" );
		if ( !f ) {
			goto _exit_;
		}
		jpeg_stdio_dest( &cinfo, f );
	} else {
		dest.dst = m_buf;
		dest.buf = &out_buf;

		jpeg_buffer_dest( &cinfo, &dest );

		dest.pub.next_output_byte = &out_buf[0];
		dest.pub.free_in_buffer = out_buf.size();
	}

	if ( setjmp( jerr.setjmp_buffer ) == 0 ) {
		cinfo.image_width = width;
		cinfo.image_height = height;
		cinfo.input_components = channels;
		cinfo.in_color_space = channels > 1 ? JCS_RGB : JCS_GRAYSCALE;

		jpeg_set_defaults( &cinfo );
		jpeg_set_quality( &cinfo, quality,
						  TRUE /* limit to baseline-JPEG values */ );
		jpeg_start_compress( &cinfo, TRUE );

		if ( channels > 1 ) {
			_buffer.allocate(width * channels);
		}
		buffer = _buffer;

		for ( int y = 0; y < height; y++ ) {
			uchar* data = img.data + img.step * y, *ptr = data;

			if ( _channels == 3 ) {
				icvCvt_BGR2RGB_8u_C3R( data, 0, buffer, 0, cvSize(width, 1) );
				ptr = buffer;
			} else if ( _channels == 4 ) {
				icvCvt_BGRA2BGR_8u_C4C3R( data, 0, buffer, 0, cvSize(width, 1), 2 );
				ptr = buffer;
			}

			jpeg_write_scanlines( &cinfo, &ptr, 1 );
		}

		jpeg_finish_compress( &cinfo );
		result = true;
	}

_exit_:
	if (f) { fclose(f); }
	jpeg_destroy_compress( &cinfo );

	return result;
}

}

#endif

/* End of file. */
